model.predict() == ValueError: Error when checking input: expected flatten_input to have 3 dimension